home *** CD-ROM | disk | FTP | other *** search
/ CU Amiga Super CD-ROM 16 / CU Amiga Magazine's Super CD-ROM 16 (1997-10-16)(EMAP Images)(GB)[!][issue 1997-11].iso / CUCD / Graphics / Ghostscript / source / zfdcte.c < prev    next >
C/C++ Source or Header  |  1995-01-24  |  8KB  |  263 lines
  1. /* Copyright (C) 1994 Aladdin Enterprises.  All rights reserved.
  2.   
  3.   This file is part of Aladdin Ghostscript.
  4.   
  5.   Aladdin Ghostscript is distributed with NO WARRANTY OF ANY KIND.  No author
  6.   or distributor accepts any responsibility for the consequences of using it,
  7.   or for whether it serves any particular purpose or works at all, unless he
  8.   or she says so in writing.  Refer to the Aladdin Ghostscript Free Public
  9.   License (the "License") for full details.
  10.   
  11.   Every copy of Aladdin Ghostscript must include a copy of the License,
  12.   normally in a plain ASCII text file named PUBLIC.  The License grants you
  13.   the right to copy, modify and redistribute Aladdin Ghostscript, but only
  14.   under certain conditions described in the License.  Among other things, the
  15.   License requires that the copyright notice and this notice be preserved on
  16.   all copies.
  17. */
  18.  
  19. /* zfdcte.c */
  20. /* DCTEncode filter creation */
  21. #include "memory_.h"
  22. #include "stdio_.h"            /* for jpeglib.h */
  23. #include "jpeglib.h"
  24. #include "ghost.h"
  25. #include "errors.h"
  26. #include "oper.h"
  27. #include "idict.h"
  28. #include "idparam.h"
  29. #include "strimpl.h"
  30. #include "sdct.h"
  31. #include "sjpeg.h"
  32. #include "ifilter.h"
  33.  
  34. /* Import the common setup routines from zfdctc.c */
  35. int zfdct_setup(P2(const ref *op, stream_DCT_state *pdct));
  36. int zfdct_setup_quantization_tables(P3(const ref *op, stream_DCT_state *pdct,
  37.                        bool is_encode));
  38. int zfdct_setup_huffman_tables(P3(const ref *op, stream_DCT_state *pdct,
  39.                   bool is_encode));
  40. int zfdct_byte_params(P4(const ref *op, int start, int count, UINT8 *pvals));
  41.  
  42. /* Collect encode-only parameters. */
  43. private int
  44. dct_setup_samples(const ref *op, const char _ds *kstr, int num_colors,
  45.           jpeg_compress_data *jcdp, bool is_vert)
  46. {    int code;
  47.     int i;
  48.     ref *pdval;
  49.     jpeg_component_info * comp_info = jcdp->cinfo.comp_info;
  50.     UINT8 samples[4];
  51.     /* Adobe default is all sampling factors = 1,
  52.      * which is NOT the IJG default, so we must always assign values.
  53.      */
  54.     if ( op != 0 && dict_find_string(op, kstr, &pdval) > 0 )
  55.     {    if ( r_size(pdval) < num_colors )
  56.             return_error(e_rangecheck);
  57.         if ( (code = zfdct_byte_params(pdval, 0, num_colors, samples)) < 0 )
  58.             return code;
  59.     }
  60.     else
  61.     {    samples[0] = samples[1] = samples[2] = samples[3] = 1;
  62.     }
  63.     for ( i = 0; i < num_colors; i++ )
  64.     {    if ( samples[i] < 1 || samples[i] > 4 )
  65.             return_error(e_rangecheck);
  66.         if ( is_vert )
  67.             comp_info[i].v_samp_factor = samples[i];
  68.         else
  69.             comp_info[i].h_samp_factor = samples[i];
  70.     }
  71.     return 0;
  72. }
  73.  
  74. private int
  75. zfdcte_setup(const ref *op, stream_DCT_state *pdct)
  76. {    jpeg_compress_data *jcdp = pdct->data.compress;
  77.     uint Columns, Rows, Resync;
  78.     int num_colors;
  79.     int Blend;
  80.     ref *mstr;
  81.     int i;
  82.     int code;
  83.  
  84.     /* Required parameters for DCTEncode.
  85.      * (DCTDecode gets the equivalent info from the SOF marker.)
  86.      */
  87.     if ( (code = dict_uint_param(op, "Columns", 1, 0xffff, 0,
  88.                      &Columns)) < 0 ||
  89.          (code = dict_uint_param(op, "Rows", 1, 0xffff, 0,
  90.                      &Rows)) < 0 ||
  91.          (code = dict_int_param(op, "Colors", 1, 4, -1,
  92.                     &num_colors)) < 0
  93.        )
  94.         return code;
  95.     /* Set up minimal image description & call set_defaults */
  96.     jcdp->cinfo.image_width = Columns;
  97.     jcdp->cinfo.image_height = Rows;
  98.     jcdp->cinfo.input_components = num_colors;
  99.     switch ( num_colors )
  100.     {
  101.     case 1:
  102.         jcdp->cinfo.in_color_space = JCS_GRAYSCALE;
  103.         break;
  104.     case 3:
  105.         jcdp->cinfo.in_color_space = JCS_RGB;
  106.         break;
  107.     case 4:
  108.         jcdp->cinfo.in_color_space = JCS_CMYK;
  109.         break;
  110.     default:
  111.         jcdp->cinfo.in_color_space = JCS_UNKNOWN;
  112.     }
  113.     if ( (code = gs_jpeg_set_defaults(pdct)) < 0 )
  114.         return code;
  115.     /* Change IJG colorspace defaults as needed;
  116.      * set ColorTransform to what will go in the Adobe marker.
  117.      */
  118.     switch ( num_colors )
  119.     {
  120.     case 3:
  121.         if ( pdct->ColorTransform < 0 )
  122.             pdct->ColorTransform = 1; /* default */
  123.         if ( pdct->ColorTransform == 0 )
  124.         {
  125.             if ( (code = gs_jpeg_set_colorspace(pdct, JCS_RGB)) < 0 )
  126.                 return code;
  127.         }
  128.         else
  129.             pdct->ColorTransform = 1; /* flag YCC xform */
  130.         break;
  131.     case 4:
  132.         if ( pdct->ColorTransform < 0 )
  133.             pdct->ColorTransform = 0; /* default */
  134.         if ( pdct->ColorTransform != 0 )
  135.         {    if ( (code = gs_jpeg_set_colorspace(pdct, JCS_YCCK)) < 0 )
  136.                 return code;
  137.             pdct->ColorTransform = 2; /* flag YCCK xform */
  138.         }
  139.         else
  140.         {    if ( (code = gs_jpeg_set_colorspace(pdct, JCS_CMYK)) < 0 )
  141.                 return code;
  142.         }
  143.         break;
  144.     default:
  145.         pdct->ColorTransform = 0; /* no transform otherwise */
  146.         break;
  147.     }
  148.     /* Optional encoding-only parameters */
  149.     if ( dict_find_string(op, "Markers", &mstr) > 0 )
  150.     {    check_read_type(*mstr, t_string);
  151.         pdct->Markers.data = mstr->value.const_bytes;
  152.         pdct->Markers.size = r_size(mstr);
  153.     }
  154.     if ( (code = dict_bool_param(op, "NoMarker", false,
  155.                      &pdct->NoMarker)) < 0 ||
  156.          (code = dict_uint_param(op, "Resync", 0, 0xffff, 0,
  157.                      &Resync)) < 0 ||
  158.          (code = dict_int_param(op, "Blend", 0, 1, 0,
  159.                     &Blend)) < 0 ||
  160.          (code = dct_setup_samples(op, "HSamples", num_colors,
  161.                        jcdp, false)) < 0 ||
  162.          (code = dct_setup_samples(op, "VSamples", num_colors,
  163.                        jcdp, true)) < 0
  164.         )
  165.         return code;
  166.     jcdp->cinfo.write_JFIF_header = FALSE;
  167.     jcdp->cinfo.write_Adobe_marker = FALSE;    /* must do it myself */
  168.     jcdp->cinfo.restart_interval = Resync;
  169.     /* What to do with Blend ??? */
  170.     if ( pdct->data.common->Relax == 0 )
  171.     {    jpeg_component_info *comp_info = jcdp->cinfo.comp_info;
  172.         int num_samples;
  173.         for ( i = 0, num_samples = 0; i < num_colors; i++ )
  174.             num_samples += comp_info[i].h_samp_factor *
  175.                        comp_info[i].v_samp_factor;
  176.         if ( num_samples > 10 )
  177.             return_error(e_rangecheck);
  178.         /* Note: by default the IJG software does not allow
  179.          * num_samples to exceed 10, Relax or no.  For full
  180.          * compatibility with Adobe's non-JPEG-compliant
  181.          * software, set MAX_BLOCKS_IN_MCU to 64 in jpeglib.h.
  182.          */
  183.     }
  184.     return 0;
  185. }
  186.  
  187. /* <target> <dict> DCTEncode/filter <file> */
  188. private int
  189. zDCTE(os_ptr op)
  190. {    stream_DCT_state state;
  191.     jpeg_compress_data *jcdp;
  192.     int code;
  193.     int npop;
  194.     const ref *dop;
  195.     uint dspace;
  196.     ref *pdval;
  197.  
  198.     /* First allocate space for IJG parameters. */
  199.     jcdp = gs_malloc(1, sizeof(*jcdp), "zDCTE");
  200.     if ( jcdp == 0 )
  201.         return_error(e_VMerror);
  202.     state.data.compress = jcdp;
  203.     if ( (code = gs_jpeg_create_compress(&state)) < 0 )
  204.         goto fail;    /* correct to do jpeg_destroy here */
  205.     /* Read parameters from dictionary */
  206.     if ( (code = zfdct_setup(op, &state)) < 0 )
  207.         goto fail;
  208.     npop = code;
  209.     if ( npop == 0 )
  210.       dop = 0, dspace = 0;
  211.     else
  212.       dop = op, dspace = r_space(op);
  213.     if ( (code = zfdcte_setup(dop, &state)) < 0 )
  214.         goto fail;
  215.     /* Check for QFactor without QuantTables. */
  216.     if ( dop == 0 || dict_find_string(dop, "QuantTables", &pdval) <= 0 )
  217.       {    /* No QuantTables, but maybe a QFactor to apply to default. */
  218.         if ( state.QFactor != 1.0 )
  219.         {    code = gs_jpeg_set_linear_quality(&state,
  220.                     (int) (min(state.QFactor, 100.0)
  221.                            * 100.0 + 0.5),
  222.                     TRUE);
  223.             if ( code < 0 )
  224.                 return code;
  225.         }
  226.  
  227.       }
  228.     if ( (code = zfdct_setup_huffman_tables(dop, &state, true)) < 0 ||
  229.          (code = zfdct_setup_quantization_tables(dop, &state, true)) < 0
  230.        )
  231.         goto fail;
  232.     /* Create the filter. */
  233.     jcdp->template = s_DCTE_template;
  234.     /* Make sure we get at least a full scan line of input. */
  235.     state.scan_line_size = jcdp->cinfo.input_components *
  236.                    jcdp->cinfo.image_width;
  237.     jcdp->template.min_in_size =
  238.         max(s_DCTE_template.min_in_size, state.scan_line_size);
  239.     /* Make sure we can write the user markers in a single go. */
  240.     jcdp->template.min_out_size =
  241.         max(s_DCTE_template.min_out_size, state.Markers.size);
  242.     code = filter_write(op, npop, &jcdp->template,
  243.                 (stream_state *)&state, dspace);
  244.     if ( code >= 0 )        /* Success! */
  245.         return code;
  246.     /* We assume that if filter_write fails, the stream has not been
  247.      * registered for closing, so s_DCTE_release will never be called.
  248.      * Therefore we free the allocated memory before failing.
  249.      */
  250.  
  251. fail:
  252.     gs_jpeg_destroy(&state);
  253.     gs_free(jcdp, 1, sizeof(*jcdp), "zDCTE fail");
  254.     return code;
  255. }
  256.  
  257. /* ------ Initialization procedure ------ */
  258.  
  259. BEGIN_OP_DEFS(zfdcte_op_defs) {
  260.         op_def_begin_filter(),
  261.     {"2DCTEncode", zDCTE},
  262. END_OP_DEFS(0) }
  263.